#ifndef AGX_H
#define AGX_H
#include <iostream>
#include <fstream>
// AGX类
class AGX
{
private:
    std::string model; // 型号
    int aiPerformance; // AI性能（TOPS）。用TOPS（每秒万亿次运算）来衡量AI处理器的性能，也称之为“算力”
    int cudaCores;     // CUDA核心数
    int tensorCores;   // Tensor CORE数
    int memory;        // 显存（GB）
    int storage;       // 存储（GB）

public:
    AGX();
    void setModel(const std::string &model);
    void setAiPerformance(int aiPerformance);
    void setCudaCores(int cudaCores);

    void setTensorCores(int tensorCores);
    void setMemory(int memory);

    void setStorage(int storage);
    // 屏幕输出
    void print();
    // 本地文件保存
    void save(const std::string &filename);
};

#endif